from llm.ChatGLM import ChatGLM
from llm.ZhipuAI import ZhipuAI

llms ={
        "ChatGLM3-6B":[False,ChatGLM],
            "ZhipuAI":[True,ZhipuAI]
            }

class LLMFactory:
    @staticmethod   
    def get( name :str= None ):
        if name == None:
            return llms["ChatGLM3-6B"]
        
        if name in llms.keys():
            return llms[name]
        else:
            return llms["ChatGLM3-6B"]